from .vq_trainer import VQGANTrainer